package com.allwees.bs.module.pms.util;

import com.allwees.bs.module.pms.dto.ProductDto;
import com.allwees.bs.module.pms.entity.Category;
import com.allwees.bs.module.pms.entity.Product;
import com.allwees.bs.module.pms.entity.Sku;
import com.allwees.bs.module.pms.vo.ExportProductVo;
import com.allwees.bs.module.pms.vo.ExportSkuVo;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

public class ProductExportUtil {
	public static List<ExportProductVo> toExportVo(List<ProductDto> products) {
		List<ExportProductVo> vos = new ArrayList<>();
		products.forEach(product -> vos.add(new ExportProductVo(product)));
		return vos;
	}

	public static List<ExportSkuVo> skuToExportVo(Map<String, Category> categoryMap, Map<String, Product> productMap, List<Sku> skus) {
		List<ExportSkuVo> vos = new ArrayList<>();
		skus.forEach(sku -> {
			Product product = productMap.get(sku.getProductUuid());
			vos.add(new ExportSkuVo(categoryMap.get(product.getCategoryNo()), product, sku));
		});
		return vos;
	}
}
